package com.andnnl.imageutils;

import java.awt.*;
import java.awt.image.*;
import java.awt.font.*;
import java.awt.geom.*;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.UUID;

import javax.imageio.ImageIO;
/**
 * Created by bdfus004 on 2018/1/22.
 */
public class ChartGraphics {
    // 默认格式
    private static final String FORMAT_NAME = "png";
    // 默认 宽度
    private static final int WIDTH = 400;
    // 默认 高度
    private static final int HEIGHT =400;
    // 水印透明度
    private static float alpha = 1.0f;

    // 水印横向位置
    private static int positionWidth = 40;
    // 水印纵向位置
    private static int positionHeight = 240;
    /**
     * 创建图片
     * @param content 内容
     * @param font  字体
     * @param width 宽
     * @param height 高
     * @return
     */
    private static BufferedImage createImage(String content,Font font,Integer width,Integer height){
        Integer degree = -45;
        BufferedImage bi = new BufferedImage(width, height, BufferedImage.TRANSLUCENT);
        Graphics2D g2 = (Graphics2D)bi.getGraphics();
        g2.setBackground(Color.WHITE);
        g2.clearRect(0, 0, width, height);
        g2.setPaint(Color.BLACK);
        g2.setFont(font);
        // 4、设置水印旋转
        if (null != degree) {
            g2.rotate(Math.toRadians(degree),(double) bi.getWidth() / 2, (double) bi.getHeight() / 2);
        }
        // 5、设置水印文字颜色
        g2.setColor(Color.black);
        // 7、设置水印文字透明度
        g2.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP,alpha));
        // 8、第一参数->设置的内容，后面两个参数->文字在图片上的坐标位置(x,y)
        g2.drawString(content, positionWidth, positionHeight);

        return bi;
    }

    /**
     * 获取 图片
     * @param content 内容
     * @param font  字体
     * @param width 宽
     * @param height 高
     * @return
     */
    public static BufferedImage getImage(String content,Font font,Integer width,Integer height){
        width=width==null?WIDTH:width;
        height=height==null?HEIGHT:height;
        if(null==font)
            font = new Font("宋体", Font.BOLD, 80);
        return createImage(content, font, width, height);
    }

    /**
     * 获取 图片
     * @param content 内容
     * @param width 宽
     * @param height 高
     * @return
     */
    public static BufferedImage getImage(String content,Integer width,Integer height){

        return getImage(content, null,width, height);
    }

    /**
     * 获取图片
     * @param content 内容
     * @return
     */
    public static BufferedImage getImage(String content){

        return getImage(content, null, null);
    }

    /**
     *  获取图片
     * @param content 内容
     * @param font 字体
     * @param width 宽
     * @param height 高
     * @param destPath 输出路径
     * @throws IOException
     */
    public static void getImage(String content,Font font ,Integer width,Integer height,String destPath) throws IOException{
        mkdirs(destPath);
        String file = UUID.randomUUID().toString()+".jpg";
        ImageIO.write(getImage(content,font,width,height),FORMAT_NAME, new File(destPath+"/"+file));
    }

    /**
     * 获取图片
     * @param content 内容
     * @param font 字体
     * @param width 宽
     * @param height 高
     * @param output 输出流
     * @throws IOException
     */
    public static void getImage(String content,Font font,Integer width,Integer height, OutputStream output) throws IOException{
        ImageIO.write(getImage(content,font,width,height), FORMAT_NAME, output);
    }

    /**
     * 获取图片
     * @param content 内容
     * @param width 宽
     * @param height 高
     * @param destPath 输出路径
     * @throws IOException
     */
    public static void getImage(String content,Integer width,Integer height,String destPath) throws IOException{
        getImage(content, null, width, height, destPath);
    }

    /**
     * 获取图片
     * @param content 内容
     * @param width 宽
     * @param height 高
     * @param output 输出流
     * @throws IOException
     */
    public static void getImage(String content,Integer width,Integer height, OutputStream output) throws IOException{
        getImage(content, null, width, height, output);
    }


    /**
     * 创建 目录
     * @param destPath
     */
    public static void mkdirs(String destPath) {
        File file =new File(destPath);
        //当文件夹不存在时，mkdirs会自动创建多层目录，区别于mkdir．(mkdir如果父目录不存在则会抛出异常)
        if (!file.exists() && !file.isDirectory()) {
            file.mkdirs();
        }
    }

    public static void main(String[] args) throws Exception {
        getImage("发展小贷", 400, 400, "C:\\Users\\bdfus004\\Desktop\\test");

    }

    public static byte[] bufferedImage2ByteArray(BufferedImage image) throws IOException{
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        ImageIO.write(image, "jpeg", baos);
        return baos.toByteArray();
    }
}
